package edu.uta.cfl.GUI;
/**
 * State Pattern - Context class
 * Based on the test strength value, this class will make calls to appropriate concrete class.
 * @author jagan
 *
 */
public class ContextClass {

	private int strength;
	private String[] columnHeader;

	public ContextClass(int strength) {
		super();
		this.strength = strength;

	}

	public String[] getColumnHeader()
	{
		if(strength == 2)
		{
			columnHeader = new TestStrength2().getColumnName();
		}
		if(strength == 3)
		{
			columnHeader = new TestStrength3().getColumnName();
		}
		if(strength == 4)
		{
			columnHeader = new TestStrength4().getColumnName();
		}
		if(strength == 5)
		{
			columnHeader = new TestStrength5().getColumnName();
		}
		if(strength == 6)
		{
			columnHeader = new TestStrength6().getColumnName();
		}

		return columnHeader;

	}

}
